package jjn.round1;

/**
 * @author Jiang Jining
 * @since 2023-05-28 15:22
 */
public class LeetCode309_BestTimeToBuyAndSellStockWithCoolDown {
    public int maxProfit(int[] prices) {
        // 状态包含：持股/卖出且冷冻/卖出非冷冻
        int[][] dp = new int[prices.length][3];
        dp[0][0] = -prices[0];
        dp[0][1] = 0;
        dp[0][2] = 0;
        for (int i = 1; i < prices.length; i++) {
            // 1.保留昨天买的股票 2. 卖出非冷冻期的股票
            dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][2] - prices[i]);
            // 卖出昨天购买的股票
            dp[i][1] = dp[i - 1][0] + prices[i];
            // 1.昨天卖出,非冷冻 2. 昨天没卖出，非冷冻
            dp[i][2] = Math.max(dp[i - 1][1], dp[i - 1][2]);
        }
        // 最后一天依然持有股票,显然不是最佳解
        return Math.max(dp[prices.length - 1][1], dp[prices.length - 1][2]);
    }
}
